package github.shuhai65.chatbot.domain.zsxq.service;


import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import github.shuhai65.chatbot.domain.zsxq.IZsxqApi;
import github.shuhai65.chatbot.domain.zsxq.model.aggregates.TopicAggregates;
import github.shuhai65.chatbot.domain.zsxq.model.req.AnswerReq;
import github.shuhai65.chatbot.domain.zsxq.model.req.ReqData;
import github.shuhai65.chatbot.domain.zsxq.model.res.AnswerRes;
import org.apache.http.HttpStatus;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.util.EntityUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Service;

import java.io.IOException;

@Service
public class ZsxqApi implements IZsxqApi {

    private Logger logger = LoggerFactory.getLogger(ZsxqApi.class);


    @Override
    public TopicAggregates queryTopics(String groupId, String cookie) throws IOException {
        CloseableHttpClient httpClient = HttpClientBuilder.create().build();
        HttpGet httpGet = new HttpGet("https://api.zsxq.com/v2/groups/"+groupId+"/topics?scope=all&count=20");
        httpGet.addHeader("cookie",cookie);
        httpGet.addHeader("content-type","application/json; charset=UTF-8");
        CloseableHttpResponse response = httpClient.execute(httpGet);
        if(response.getStatusLine().getStatusCode() == HttpStatus.SC_OK){
            String jsonStr = EntityUtils.toString(response.getEntity());
            logger.info("拉取文章。groupId：{} jsonStr：{}", groupId, jsonStr);
            return JSON.parseObject(jsonStr, TopicAggregates.class);
        }else{
            throw new RuntimeException("queryTopics Err Code is " + response.getStatusLine().getStatusCode());
        }
    }

    @Override
    public boolean answer(String groupId, String cookie, String topicId, String text) throws IOException {
        CloseableHttpClient httpClient = HttpClientBuilder.create().build();
        HttpPost httpPost = new HttpPost("https://api.zsxq.com/v2/topics/"+topicId+"/comments");
        httpPost.addHeader("cookie",cookie);
        httpPost.addHeader("content-type","application/json; charset=UTF-8");
        httpPost.addHeader("user-agent","Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36");
        String paramJson = "{\n" +
                "  \"req_data\": {\n" +
                "    \"text\": \""+text+"\\n\",\n" +
                "    \"image_ids\": [],\n" +
                "    \"mentioned_user_ids\": []\n" +
                "  }\n" +
                "}";

        AnswerReq answerReq = new AnswerReq(new ReqData(text));
        //String paramJson = JSONObject.toJSONString(answerReq);
        StringEntity stringEntity = new StringEntity(paramJson, ContentType.create("text/json", "UTF-8"));
        httpPost.setEntity(stringEntity);
       // StringEntity stringEntity = new StringEntity(paramJson, ContentType.create("text/json", "UTF-8").getCharset());
       // httpPost.setEntity(stringEntity);
        CloseableHttpResponse response = httpClient.execute(httpPost);
        if(response.getStatusLine().getStatusCode() == HttpStatus.SC_OK){
            String jsonStr = EntityUtils.toString(response.getEntity());
            logger.info("回答结果。groupId：{} topicId：{} jsonStr：{}", groupId, topicId, jsonStr);
            AnswerRes answerRes = JSON.parseObject(jsonStr, AnswerRes.class);
            return answerRes.isSucceeded();
        }else{
            throw new RuntimeException("answer Err Code is " + response.getStatusLine().getStatusCode());
        }
    }
}
